package cn.t.repository.impl.device;

import cn.t.converter.device.CargowayConverter;
import cn.t.core.repoistory.RepositoryImpl;
import cn.t.model.device.CargowayDO;
import cn.t.persistence.entity.device.CargowayPO;
import cn.t.persistence.mapper.device.CargowayMapper;
import cn.t.repository.device.CargowayRepository;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Repository;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StringUtils;

import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 货道表(Cargoway)数据仓库
 *
 * @author t
 * @since 2024-05-24 10:18:08
 */
@Slf4j
@Repository
@AllArgsConstructor
public class CargowayRepositoryImpl extends RepositoryImpl<CargowayMapper, CargowayPO, CargowayConverter, CargowayDO> implements CargowayRepository {


    @Override
    public List<CargowayDO> findByModel(CargowayDO cargowayDO, int start, Integer pageSize, String field, String direction) {

        String lastSql = StringUtils.hasText(field) ? " order by " + field + " " + direction + " limit " + start + "," + pageSize : " limit " + start + "," + pageSize;

        List<CargowayPO> list = getWrapper(cargowayDO).last(lastSql).list();
        if (CollectionUtils.isEmpty(list)) {
            return Collections.emptyList();
        }

        return list.stream().map(converter::po2Model).collect(Collectors.toList());
    }

    private LambdaQueryChainWrapper<CargowayPO> getWrapper(CargowayDO cargowayDO) {
        LambdaQueryChainWrapper<CargowayPO> wrapper = lambdaQuery();
        if (!ObjectUtils.isEmpty(cargowayDO.getDeviceId())) {
            wrapper.eq(CargowayPO::getDeviceId, cargowayDO.getDeviceId());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getCargowayNo())) {
            wrapper.eq(CargowayPO::getCargowayNo, cargowayDO.getCargowayNo());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getCargowayVol())) {
            wrapper.eq(CargowayPO::getCargowayVol, cargowayDO.getCargowayVol());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getStatus())) {
            wrapper.eq(CargowayPO::getStatus, cargowayDO.getStatus());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getGravityScaleNo())) {
            wrapper.eq(CargowayPO::getGravityScaleNo, cargowayDO.getGravityScaleNo());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getDoorLockNo())) {
            wrapper.eq(CargowayPO::getDoorLockNo, cargowayDO.getDoorLockNo());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getLedNo())) {
            wrapper.eq(CargowayPO::getLedNo, cargowayDO.getLedNo());
        }
        if (!ObjectUtils.isEmpty(cargowayDO.getMotorNo())) {
            wrapper.eq(CargowayPO::getMotorNo, cargowayDO.getMotorNo());
        }
        wrapper.in(!CollectionUtils.isEmpty(cargowayDO.getDeviceIds()), CargowayPO::getDeviceId, cargowayDO.getDeviceIds());
        return wrapper;
    }

    @Override
    public long count(CargowayDO cargowayDO) {
        return getWrapper(cargowayDO).count();
    }

    @Override
    public void addOrUpdateBatch(List<CargowayDO> cargowayDOS) {

        if (CollectionUtils.isEmpty(cargowayDOS)) {
            return;
        }

        List<CargowayPO> cargowayPOS = cargowayDOS.stream().map(this.converter::model2Po).collect(Collectors.toList());
        this.saveOrUpdateBatch(cargowayPOS);
    }

    @Override
    public void update(CargowayDO cargowayDO, List<Long> cargowayIds) {
        LambdaUpdateWrapper<CargowayPO> wrapper = new LambdaUpdateWrapper<>();
        wrapper.set(!ObjectUtils.isEmpty(cargowayDO.getCargowayVol()), CargowayPO::getCargowayVol, cargowayDO.getCargowayVol());
        wrapper.in(CargowayPO::getId, cargowayIds);
        this.update(wrapper);
    }

}
